how to work with matrices in numpy